To install PyTorch with GPU support visit this link. Select Version, OS, Language, package installer, CUDA version and then follow the highlighted portion of the following image to install. Now verify your installation using these python scripts. All the libraries are installed. Now turn your CREATIVE mode on.

When it comes to working with deep learning Python I highly recommend that you use a Linux environment. Deep learning tools can be more easily configured and installed on Linux, allowing you to develop and run neural networks quickly. Of course, configuring your own deep learning Python Linux development environment can be quite the tedious task, especially if you are new to Linux, a beginner at working the command line/terminal, or a novice when compiling and installing packages by hand. In order to help you jump start your deep learning Python education, I have created an Ubuntu virtual machine with all necessary deep learning libraries you need to successful (including Keras, TensorFlow, scikit-learn, scikit-image, OpenCV, and others) pre-configured and pre-installed. This virtual machine is part of all three bundles of my book, Deep Learning for Computer Vision with Python.